/*
异步缓冲区类的设计
*/

#ifndef __M_BUFFER_H__
#define __M_BUFFER_H__

#include <iostream>
#include <vector>
#include <cassert>

#define DEFAULT_BUFFER_SIZE (1 * 1024 * 1024)
#define THREHOLD_BUFFER_SIZE (8 * 1024 * 1024)
#define LINEARGROWTH_BUFFER_SIZE (1 * 1024 * 1024)

namespace clog
{
    class Buffer
    {
    public:
        Buffer()
            : _buffer(DEFAULT_BUFFER_SIZE), _reader_idx(0), _writer_idx(0)
        {
        }
        // 写入数据
        void push(const char *data, size_t len)
        {
            // 缓冲区满了：1.返回阻塞 2.扩容
            // if (len > writeAbleSize())s
            //     return;
            // 扩容--用于极限测试
            ensureEnoughSize(len);
            // 将数据拷贝到缓冲区
            std::copy(data, data + len, &_buffer[_writer_idx]);
            // 当前写入位置向后偏移
            moveWriterIndx(len);
        }
        // 当前可写入的空间大小
        size_t writeAbleSize()
        {
            // 对于扩容来说，不存在可写入空间的大小
            // 该接口只针对缓冲区大小固定的场景提供
            return (_buffer.size() - _writer_idx); // 因为是双缓冲区机制，写满就交换缓冲区，所以在单缓冲区中不存在空间循环利用的情况
        }
        // 读取数据的起始位置
        const char *begin()
        {
            return &_buffer[_reader_idx];
        }
        // 可读数据的起始长度
        size_t readAbleSize()
        {
            return (_writer_idx - _reader_idx);
        }
        // 当前读取位置向后偏移
        void movReaderIndx(size_t len)
        {
            // 判断是否有足够的数据被读取
            assert(len <= readAbleSize());
            _reader_idx += len;
        }
        // 重置读写位置，初始化缓冲区
        void reset()
        {
            _reader_idx = 0;
            _writer_idx = 0;
        }
        // 两个缓冲区的交换
        void swap(Buffer &buffer)
        {
            _buffer.swap(buffer._buffer);
            std::swap(_reader_idx, buffer._reader_idx);
            std::swap(_writer_idx, buffer._writer_idx);
        }
        // 判断缓冲区是否为空
        bool empty()
        {
            return (_reader_idx == _writer_idx);
        }

    private:
        // 当前写入位置向后偏移
        void moveWriterIndx(size_t len)
        {
            assert((_writer_idx + len) <= _buffer.size());
            _writer_idx += len;
        }

        void ensureEnoughSize(size_t len)
        {
            // 不需要扩容
            if (len <= writeAbleSize())
                return;
            size_t newsize = 0;
            if (_buffer.size() < THREHOLD_BUFFER_SIZE)
            {
                // 小于阈值 二倍扩容
                newsize = _buffer.size() * 2 + len;
            }
            else
            {
                // 大于阈值 线性增长
                newsize = _buffer.size() + LINEARGROWTH_BUFFER_SIZE + len;
            }
            _buffer.resize(newsize);
        }

        std::vector<char> _buffer;
        size_t _reader_idx; // 可读位置的指针
        size_t _writer_idx; // 可写位置的指针
    };
}
#endif